Compute numpy arrays based on another numpy array with each other